package com.tianzehua.netty.chatexample;

import io.netty.channel.Channel;
import io.netty.channel.ChannelHandlerContext;
import io.netty.channel.SimpleChannelInboundHandler;
import io.netty.channel.group.ChannelGroup;
import io.netty.channel.group.DefaultChannelGroup;
import io.netty.util.concurrent.GlobalEventExecutor;

import java.time.LocalDateTime;


/**
 * chat server handler {@link SimpleChannelInboundHandler}
 * @author tianzehua
 * @date 2019/03/08
 */
public class MyChatServerHandler extends SimpleChannelInboundHandler<String> {


    /**
     * ChannelGroup
     * 定义一个 ChannelGroup 的成员变量，成员变量是作用在类上的变量，初始必须有值，存在堆内存中，局部变量存在栈内存中
     */
    private  static ChannelGroup channelGroup = new DefaultChannelGroup(GlobalEventExecutor.INSTANCE);

    @Override
    protected void channelRead0(ChannelHandlerContext ctx, String msg) {
        /* 服务端收到任意的一个客户端发来的消息的时候，会触发channelRead0这个方法来进行回调 */
        Channel channel = ctx.channel();
        /* 在接收到的服务器的数据中，因为是广播，所以会收到所有的客户端发来的消息，
        当消息是自己发出去的那条时，前面加“自己”，否则都是所发消息客服端的地址 */
        channelGroup.forEach(ch ->{
            if (channel != ch){
                ch.writeAndFlush(channel.remoteAddress() + "发送的消息：" + LocalDateTime.now() + ":" + msg + "\n");
            }else{
                ch.writeAndFlush("【自己】" + msg + "\n");
            }
        });
    }

    @Override
    public void handlerAdded(ChannelHandlerContext ctx)  {
        /* 表示客户端与服务器端已经建立好了连接 */
        Channel channel = ctx.channel();
        /* 当建立好连接后，就可以向网络广播该连接的用户上线的通知 ,channelGroup 可以储存所有的连接信息，用于广播的发送 */
        channelGroup.writeAndFlush("【服务器】-"+ channel.remoteAddress() + "加入:" + LocalDateTime.now() +"\n");
        channelGroup.add(channel);
    }

    @Override
    public void handlerRemoved(ChannelHandlerContext ctx)  {
        Channel channel = ctx.channel();
        channelGroup.writeAndFlush("【服务器】-"+ channel.remoteAddress() + "离开\n");
        /* 当客户端断开连接时，netty 会自动调用下面的代码 */
        /*channelGroup.remove(channel);*/
    }

    @Override
    public void channelActive(ChannelHandlerContext ctx)  {
        /* 连接处于活动状态 */
        Channel channel = ctx.channel();
        System.out.println(channel.remoteAddress() + "上线！");
    }

    @Override
    public void channelInactive(ChannelHandlerContext ctx)  {
        /* 连接处于非活动状态 */
        Channel channel = ctx.channel();
        System.out.println(channel.remoteAddress() + "下线！");
    }

    @Override
    public void exceptionCaught(ChannelHandlerContext ctx, Throwable cause) {
        cause.getStackTrace();
        ctx.channel().close();
    }
}
